LIHAT SEMUA: Python lambda square